How to choose right number of partitions